How to take integer as input in python